package com.shujia.profile.info.platform

import com.shujia.profile.UserProfile
import com.shujia.util.SparkUtil
import org.apache.spark.sql.{DataFrame, SparkSession}

class UserProfileMemberId extends UserProfile {

  flagName = "memberId"

  /**
   * 用于提取用户画像的接口
   *
   * @param spark ： spark环境
   * @return
   */
  override def extractProfile(spark: SparkSession): DataFrame = {

    import spark.implicits._

    spark
      .table("dim.dim_pub_t_member")
      .select($"id", $"id" as "memberId")
  }
}

object UserProfileMemberId {
  def main(args: Array[String]): Unit = {
    //获取参数,在提交任务时，在包名后面增加的参数
    val ds: String = args.head
    //spark环境
    val spark: SparkSession = SparkUtil.getSession()

    new UserProfileMemberId().run(spark, ds)
  }
}
